package com.tanklab.signature.common.util;

import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.data.redis.core.RedisTemplate;
import org.springframework.stereotype.Component;

import java.util.concurrent.TimeUnit;

/**
 * Token黑名单管理器
 * 使用Redis存储已登出或失效的Token
 */
@Component
public class TokenBlacklistManager {

    // Redis键的前缀，用于区分不同类型的数据
    private static final String TOKEN_BLACKLIST_PREFIX = "token:blacklist:";

    @Autowired
    private RedisTemplate<String, Object> redisTemplate;

    /**
     * 将token加入黑名单
     * @param token 需要失效的token
     * @param expireTime token的过期时间戳
     */
    public void addToBlacklist(String token, long expireTime) {
        String key = TOKEN_BLACKLIST_PREFIX + token;
        // 计算距离过期还有多少毫秒
        long ttl = expireTime - System.currentTimeMillis();

        if (ttl > 0) {
            // 将token放入Redis，并设置与token相同的过期时间
            redisTemplate.opsForValue().set(key, "1", ttl, TimeUnit.MILLISECONDS);
        }
    }

    /**
     * 检查token是否在黑名单中
     * @param token 待检查的token
     * @return 是否在黑名单中
     */
    public boolean isBlacklisted(String token) {
        String key = TOKEN_BLACKLIST_PREFIX + token;
        return Boolean.TRUE.equals(redisTemplate.hasKey(key));
    }

}